Pytorch matmul - RuntimeError: addmm_impl_cpu_ not implemented for 'Half'